package com.bigdata.spark.core.wc

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

object WordCount1 {
  def main(args: Array[String]): Unit = {
    // Spark框架 ---- spark 上下文对象 SparkContext()
    // TODO 建立和Spark框架的连接
    // JDBC : Connection
    val sparkConf = new SparkConf().setMaster("local").setAppName("WordCount")
    val sc = new SparkContext(sparkConf)

    // TODO 执行业务操作

    // 1. 读取文件，获取一行一行的数据
    // hello world
    val line: RDD[String] = sc.textFile("datas/1.txt")

    // 2. 将一行数据进行拆分，形成一个个的单词（分词）
    //    扁平化: 将整体拆分成个体的操作
    // “hello world”  => hello, world
    val words: RDD[String] = line.flatMap(_.split(" "))

    // 3. 将数据根据单词进行分组，便于统计
    // (hello, hello, hello), (world, world)
    val wordGroup: RDD[(String, Iterable[String])] = words.groupBy(word => word)

    // 4. 对分组后的数据进行转换
    // (hello, hello, hello), (world, world)
    // (hello,3) ,  (world, 2)
    val wordToCount: RDD[(String, Int)] = wordGroup.map {
      case (word, list) => (word, list.size)
    }

    // 5. 将转换的结果打印到控制台
    val array: Array[(String, Int)] = wordToCount.collect()

    array.foreach(println)

    // TODO 关闭连接
    sc.stop()
  }
}
